package com.hadoop.mr.flow;

import org.apache.hadoop.io.Text;
import org.apache.hadoop.mapreduce.Partitioner;


/**
 * 1.自定义分区类，继承 Partitioner
 * 2.第一个泛型类型对应Mapper输出key类型
 * 3.第二个泛型类型对应Mapper输出value类型
 * @author ysq
 *
 */
public class FlowPartitioner extends Partitioner<Text, Flow>{
	/*
	 * return的数字是分区编号，比如是三个分区，编号是0~2
	 * 
	 */
	@Override
	public int getPartition(Text key, Flow value, int numPartitions) {
		if(value.getAddr().equals("bj")){
			return 0;
		}else if(value.getAddr().equals("sh")){
			return 1;
		}else{
			return 2;
		}
		
	}

}
